from autogen import AssistantAgent, UserProxyAgent

# 配置LLM模型，例如使用GPT-4
config_list = [{"model": "gpt-4", "api_key": "你的OPENAI_API_KEY"}]

# 创建助理智能体
assistant = AssistantAgent(name="助理", llm_config={"config_list": config_list})

# 创建用户代理智能体（代表用户，可以执行代码）
user_proxy = UserProxyAgent(name="用户", code_execution_config={"work_dir": "coding"})

# 用户代理发起一个任务，开启对话
user_proxy.initiate_chat(
    assistant,
    message="请写一段Python代码，计算并打印出1到100之间所有奇数的和。"
)
